package com.caidan.diagram.common;

public class UserInfoHolder {
    // 创建一个 ThreadLocal 实例，用于存储用户 ID
    private static final ThreadLocal<Long> userThreadLocal = new ThreadLocal<>();

    // 设置用户 ID 的方法
    public static void setUserId(Long userId) {
        userThreadLocal.set(userId);
    }

    // 获取用户 ID 的方法
    public static Long getUserId() {
        return userThreadLocal.get();
    }

    // 清除用户 ID 的方法，防止内存泄漏
    public static void clear() {
        userThreadLocal.remove();
    }
}    